#include <iostream>
#include <pthread.h> //原生线程库
#include <string>
#include <vector>
#include <unistd.h>

void WaitThreads(const std::vector<pthread_t> &tids)
{
    for (auto &tid : tids)
    {
        pthread_join(tid, nullptr); // 回收所有线程
    }
}

pthread_cond_t gcond = PTHREAD_COND_INITIALIZER;    // 创建全局或静态条件变量
pthread_mutex_t gmutex = PTHREAD_MUTEX_INITIALIZER; // 创建全局或静态互斥锁

void *SlaverCore(void *args)
{
    std::string name = static_cast<char *>(args); // 强转回来
    // 将打印到屏幕这个共同资源进行保护，实现同步与互斥
    while (true)
    {
        // 加锁实现互斥, 条件变量一般放在锁内部即加锁和解锁之间
        pthread_mutex_lock(&gmutex);                              // 加锁
        pthread_cond_wait(&gcond, &gmutex);                       // 阻塞睡眠插入到队列， 为什么传锁，前一半释放，实现同步顺序
        std::cout << "当前被叫醒的线程是: " << name << std::endl; // 一旦被唤醒，享受打印到屏幕的资源，加锁【后一半】继续保持互斥，执行完解锁
        pthread_mutex_unlock(&gmutex);                            // 解锁
        // sleep(1);
    }
}

void *MasterCore(void *args)
{
    sleep(3);
    std::string name = static_cast<char *>(args); // 强转回来
    std::cout << "主控线程开始工作....." << std::endl;
    // 主控线程来进行唤醒队列头部线程
    while (true)
    {
        // pthread_cond_broadcast(&gcond);// 唤醒队列中所有的线程
        pthread_cond_signal(&gcond); // 铃铛唤醒一个队列头部一个线程
        std::cout << "master 唤醒一个线程...." << std::endl;
        // std::cout << name << std::endl;
        sleep(1);
    }
}

void StartSlaver(std::vector<pthread_t> *tids, int threadnum = 3)
{
    // 创建一批线程
    for (int i = 0; i < threadnum; i++)
    {
        char *name = new char[64]; // 每个线程名各自分配一个独立堆空间
        snprintf(name, 64, "slaver-%d", i + 1);
        pthread_t tid;
        int n = pthread_create(&tid, nullptr, SlaverCore, name);
        if (n == 0)
        {
            std::cout << "create " << name << " success" << std::endl;
            tids->emplace_back(tid);
        }
    }
}

void StartMaster(std::vector<pthread_t> *tids)
{
    // 创建主控线程
    pthread_t tid;
    int n = pthread_create(&tid, nullptr, MasterCore, (void *)"Master thread");
    if (n == 0)
    {
        std::cout << "create Master success" << std::endl;
        tids->emplace_back(tid);
        sleep(1);
    }
}

int main()
{
    std::vector<pthread_t> tids;
    StartMaster(&tids); // 启动主控线程
    StartSlaver(&tids); // 启动其它分支线程
    WaitThreads(tids);  // 等待线程
    return 0;
}